package com.hfzy.ihk.common.util.image;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;

public class CaptchaUtils {

    private static final String ALPHABET = "1234567890";

    /**
     * 生成一个宽为width， 高为height， 验证码为code的图片
     * @param width 图片的宽
     * @param height 图片的高
     * @param code 验证码字符串
     * @return 返回图片验证码
     */
    public static BufferedImage getImage(int width, int height, String code){
        return getImage(width, height, code, 20);
    }
    /**
     * 生成一个宽为width， 高为height， 验证码为code的图片，图片中干扰线的条数为lineCnt
     * @param width 图片的宽
     * @param height 图片的高
     * @param code 验证码字符串
     * @param lineCnt 干扰线的条数，建议为10条左右，可根据结果适当调整
     * @return 返回图片验证码
     */
    public static BufferedImage getImage(int width, int height, String code, int lineCnt){
        return createImage(width, height, code, lineCnt, 0.01);
    }
    /**
     * 生成一个宽为width， 高为height， 验证码为code的图片，图片中干扰线的条数为lineCnt
     * 噪声比为noiseRate，即图片中噪音像素点的百分比
     * @param width 图片的宽
     * @param height 图片的高
     * @param code 验证码字符串
     * @param lineCnt 干扰线的条数，建议为10条左右，可根据结果适当调整
     * @param noiseRate 图片中噪音像素点占总像素的百分比
     * @return 返回图片验证码
     */
    public static BufferedImage getImage(int width, int height, String code, int lineCnt, double noiseRate){
        return createImage(width, height, code, lineCnt, noiseRate);
    }

    /**
     *
     * 生成一个宽为width， 高为height， 验证码为code的图片，图片中干扰线的条数为lineCnt
     * 噪声比为noiseRate，即图片中噪音像素点的百分比
     * @param width 图片的宽
     * @param height 图片的高
     * @param code 验证码字符串
     * @param lineCnt 干扰线的条数，建议为10条左右，可根据结果适当调整
     * @param noiseRate 图片中噪音像素点占总像素的百分比
     * @return 返回图片验证码
     */
    private static BufferedImage createImage(int width, int height, String code, int lineCnt, double noiseRate){
        int fontWidth = ((int)(width * 0.8)) / code.length();
        int fontHeight = (int)(height * 0.7);
        //为了在任意的width和height下都能生成良好的验证码，
        //字体的大小为fontWdith何fontHeight中的小者，
        int fontSize = Math.min(fontWidth, fontHeight);
        //drawString时要用到
        int paddingX = (int) (width * 0.1);
        int paddingY = height - (height - fontSize) / 2;

        //创建图像
        BufferedImage buffimg = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
        //获得画笔
        Graphics g = buffimg.getGraphics();
        //设置画笔的颜色
        g.setColor(getRandColor(200, 255));
        //然后填充一个矩形，即设置背景色
        g.fillRect(0, 0, width, height);

        // 设置干扰线
        for (int i = 0; i < lineCnt; i++) {
            //随机获取干扰线的起点和终点
            int xs = (int)(Math.random() * width);
            int ys = (int)(Math.random() * height);
            int xe = (int)(Math.random() * width);
            int ye = (int)(Math.random() * height);
            g.setColor(getRandColor(1, 255));
            g.drawLine(xs, ys, xe, ye);
        }
        // 添加噪点
        int area = (int) (noiseRate * width * height);
        for(int i=0; i<area; ++i){
            int x = (int)(Math.random() * width);
            int y = (int)(Math.random() * height);
            buffimg.setRGB(x, y, (int)(Math.random() * 255));
        }
        //设置字体
        Font font = new Font("Ravie", Font.PLAIN, fontSize);
        g.setFont(font);

        for(int i=0; i<code.length(); ++i){
            String ch = code.substring(i, i+1);
            g.setColor(getRandColor(1, 199));
            g.drawString(ch, paddingX + fontWidth * i, paddingY);
        }
        return buffimg;

    }
    /**
     * 获取随机的颜色，r,g,b的取值在L到R之间
     * @param L 左区间
     * @param R 右区间
     * @return 返回随机颜色值
     */
    private static Color getRandColor(int L, int R){
        if(L > 255)
            L = 255;
        if(R > 255)
            R = 255;
        if(L < 0)
            L = 0;
        if(R < 0)
            R = 0;
        int interval = R - L;
        int r = L + (int)(Math.random() * interval);
        int g = L + (int)(Math.random() * interval);
        int b = L + (int)(Math.random() * interval);
        return new Color(r, g, b);
    }

    /**
     * 随机生成若干个由大小写字母和数字组成的字符串
     * @param len 随机生成len个字符
     * @return 返回随机生成的若干个由大小写字母和数字组成的字符串
     */
    public static String getRandCode(int len){
        String code = "";
        for(int i=0; i<len; ++i){
            int index = (int)(Math.random() * ALPHABET.length());
            code = code + ALPHABET.charAt(index);
        }
        return code;
    }
    /**
     * 将图片转为byte数组
     * @param image 图片
     * @return 返回byte数组
     * @throws IOException
     */
    public static byte[] getByteArray(BufferedImage image) throws IOException {
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        ImageIO.write(image, "jpeg", baos);
        return baos.toByteArray();
        //ByteArrayOutputStream 不需要close

    }
}
